import argparse
from pathlib import Path
import os

def Parse_Arguments():
    parser = argparse.ArgumentParser(description="")
    parser.add_argument('--folder_path', default="", type=str)  #文件夹路径
    parser.add_argument('--prefix', default="", type=str)  #想要在文件前加的前缀
    return parser.parse_args()

def batch_rename(folder_path, prefix):
    file_list = os.listdir(folder_path) # 获取文件夹下所有文件列表
    for i, filename in enumerate(file_list):
        if filename.endswith('.xml'): # 只处理特定的图片格式
            new_filename = f"{prefix}_{i+1}.{filename.split('.')[-1]}" # 构建新的文件名
            src = os.path.join(folder_path, filename)
            dst = os.path.join(folder_path, new_filename)
            os.rename(src, dst) # 重命名文件

def main():
    args = Parse_Arguments()
    folder_path = args.folder_path
    prefix = args.prefix
    batch_rename(folder_path, prefix)

if __name__ == "__main__":
    main()
